program for array rotation in python